#include <stdio.h>
#include <dirent.h>
#include <unistd.h>
#include <time.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <ctype.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include "server.h"

int create_socket(long port) {

    int sd, i;
    struct sockaddr_in *dir_srv = malloc(sizeof (struct sockaddr_in));
    /* Creo un descriptor de archivo. */
    if ((sd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0)
        return ERROR;

    /* Hago un bind de la direccion local */
    bzero((char *) dir_srv, sizeof (*dir_srv));
    dir_srv->sin_family = AF_INET;
    dir_srv->sin_addr.s_addr = INADDR_ANY;
    dir_srv->sin_port = htons(port);
    if ((i = bind(sd, (struct sockaddr *) dir_srv, sizeof (*dir_srv))) < 0)
        return ERROR;
    return sd;
}

int close_socket(int sd) {
    return close(sd);
}

int main(int argc, char *argv[]) {
    if (read_conf() == ERROR) {
        perror("No se ha podido leer el archivo de configuración del Servidor. Se aborta la operación.\n");
        exit(-1);
    };
    // Inicializo el servidor.
    if ((sd = create_socket(port)) < 0) {
        perror("ERROR INICIALIZANDO SERVIDOR: ");
        exit(-1);
    }
    printf("Servidor inicializado.\n");

    PACKAGE *package;
    package = malloc(sizeof (PACKAGE));
    struct sockaddr_in from_dir;
    bzero((char *) & from_dir, sizeof (from_dir));
    int fromlen = sizeof (from_dir), pid, socket = 0;

    while (1) {
        // espero petición de un cliente
        recvfrom(sd, package, sizeof (PACKAGE), 0, (struct sockaddr *) & from_dir, &fromlen);
        printf("Atendiendo nuevo cliente.\n");
        // creo proceso hijo q atienda al nuevo cliente
        if ((pid = fork()) == 0) {
            // copio el contenido del espacio de memoria de 'package' a un nuevo puntero, que es el que va a manejar el nuevo cliente
            PACKAGE *my_package = malloc(sizeof (PACKAGE));
            memcpy(my_package, package, sizeof (*package));
            // copio el contenido del espacio de memoria de 'from_dir' a una nueva estrucutra, que es la que va a manejar el nuevo cliente
            struct sockaddr_in my_from_dir = from_dir;

            int my_fromlen = fromlen, my_port = port + 1;

            // busco un nuevo puerto, a partir del siguiente al puerto de conexión
            do {
                if ((my_package->to = create_socket(my_port)) < 0) {
                    my_port++;
                    // busco hasta el puerto 65000, si no hay disponibles, vuelvo a empezar
                    if (my_port > 65000)
                        my_port = port + 1;
                }
            } while (my_package->to <= 0);

            sprintf(my_package->data, "%i", htonl(my_port));
            socket += my_package->to;
            sendto(socket, my_package, sizeof (PACKAGE), 0, (struct sockaddr *) & my_from_dir, my_fromlen);

            while (1) {
                /* Read from the socket */

                printf("---ESPERANDO COMANDO---\n");
                recvfrom(socket, my_package, sizeof (PACKAGE), 0, (struct sockaddr *) & my_from_dir, &fromlen);
                printf("---RECIBIDO COMANDO: %i\n", my_package->op);
                switch (my_package->op) {
                    case RLS:
                        rls(my_package, &my_from_dir);
                        break;
                    case RCD:
                        rcd(my_package, &my_from_dir);
                        break;
                    case GET:
                        get(my_package, &my_from_dir);
                        break;
                    case PUT:
                        put(my_package, &my_from_dir);
                        break;
                    case QUIT:
                        printf("Cerrando cliente...\n");
                        exit_client(my_package, &my_from_dir);
                        printf("Cliente cerrado...\n");
                        exit(0);
                        break;
                    default:
                        close_socket(socket);
                        exit(0);
                        break;
                }

            }
        }

    }
}
